"""
自定义登陆判定拓展视图类
"""
from django.http import JsonResponse
from django.contrib.auth.mixins import LoginRequiredMixin


class LoginRequiredJsonMixin(LoginRequiredMixin):
    """自定义LoginRequiredMixin,如果用户未登录，响应JSON，且状态码为 400 """

    def handle_no_permission(self):
        return JsonResponse({'code': 400, 'errmsg': '用户未登录！'})
